import numpy as np

# 假设有一个四维数组 
array = np.ones((2, 3, 4, 5))

# 在每个维度的开始和结束处分别填充
# 第一维：前面填充1个单位，后面填充2个单位
# 第二维：前面填充2个单位，后面填充3个单位
# 第三维：前面填充3个单位，后面填充4个单位
# 第四维：前面填充4个单位，后面填充5个单位
pad_width = ((1, 2), (2, 3), (3, 4), (4, 5))
padded_array = np.pad(array, pad_width, mode="constant", constant_values=0)

print(padded_array.shape)
